package com.sheng.leetcode.year2023.month06.day19;

import org.junit.Test;

/**
 * @author liusheng
 * @date 2023/06/19
 * <p>
 * 1262. 可被三整除的最大和<p>
 * <p>
 * 给你一个整数数组 nums，请你找出并返回能被三整除的元素最大和。<p>
 * <p>
 * 示例 1：<p>
 * 输入：nums = [3,6,5,1,8]<p>
 * 输出：18<p>
 * 解释：选出数字 3, 6, 1 和 8，它们的和是 18（可被 3 整除的最大和）。<p>
 * <p>
 * 示例 2：<p>
 * 输入：nums = [4]<p>
 * 输出：0<p>
 * 解释：4 不能被 3 整除，所以无法选出数字，返回 0。<p>
 * <p>
 * 示例 3：<p>
 * 输入：nums = [1,2,3,4,4]<p>
 * 输出：12<p>
 * 解释：选出数字 1, 3, 4 以及 4，它们的和是 12（可被 3 整除的最大和）。<p>
 * <p>
 * 提示：<p>
 * 1 <= nums.length <= 4 * 10^4<p>
 * 1 <= nums[i] <= 10^4<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/greatest-sum-divisible-by-three">1262. 可被三整除的最大和</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1262 {

    @Test
    public void test01() {
        int[] nums = {3, 6, 5, 1, 8};
//        int[] nums = {4};
//        int[] nums = {1, 2, 3, 4, 4};
        System.out.println(new Solution().maxSumDivThree(nums));
    }
}

class Solution {
    public int maxSumDivThree(int[] nums) {
        int n = nums.length;
        final int inf = 1 << 30;
        int[][] f = new int[n + 1][3];
        f[0][1] = f[0][2] = -inf;
        for (int i = 1; i <= n; ++i) {
            int x = nums[i - 1];
            for (int j = 0; j < 3; ++j) {
                f[i][j] = Math.max(f[i - 1][j], f[i - 1][(j - x % 3 + 3) % 3] + x);
            }
        }
        return f[n][0];
    }
}
